package com.sheng.leetcode.year2023.month01.day02;

import lombok.var;
import org.junit.Test;

import java.util.PriorityQueue;

/**
 * @author liusheng
 * @date 2023/01/03
 * <p>
 * 1801. 积压订单中的订单总数<p>
 * <p>
 * 给你一个二维整数数组 orders ，其中每个 orders[i] = [pricei, amounti, orderTypei] 表示有 amounti 笔类型为 orderTypei 、价格为 pricei 的订单。<p>
 * 订单类型 orderTypei 可以分为两种：<p>
 * 0 表示这是一批采购订单 buy<p>
 * 1 表示这是一批销售订单 sell<p>
 * 注意，orders[i] 表示一批共计 amounti 笔的独立订单，这些订单的价格和类型相同。对于所有有效的 i ，由 orders[i] 表示的所有订单提交时间均早于 orders[i+1] 表示的所有订单。<p>
 * 存在由未执行订单组成的 积压订单 。积压订单最初是空的。提交订单时，会发生以下情况：<p>
 * 如果该订单是一笔采购订单 buy ，则可以查看积压订单中价格 最低 的销售订单 sell 。<p>
 * 如果该销售订单 sell 的价格 低于或等于 当前采购订单 buy 的价格，则匹配并执行这两笔订单，<p>
 * 并将销售订单 sell 从积压订单中删除。否则，采购订单 buy 将会添加到积压订单中。<p>
 * 反之亦然，如果该订单是一笔销售订单 sell ，则可以查看积压订单中价格 最高 的采购订单 buy 。<p>
 * 如果该采购订单 buy 的价格 高于或等于 当前销售订单 sell 的价格，则匹配并执行这两笔订单，<p>
 * 并将采购订单 buy 从积压订单中删除。否则，销售订单 sell 将会添加到积压订单中。<p>
 * 输入所有订单后，返回积压订单中的 订单总数 。由于数字可能很大，所以需要返回对 10^9 + 7 取余的结果。<p>
 * <p>
 * 示例 1：<p>
 * 输入：orders = [[10,5,0],[15,2,1],[25,1,1],[30,4,0]]<p>
 * 输出：6<p>
 * 解释：输入订单后会发生下述情况：<p>
 * - 提交 5 笔采购订单，价格为 10 。没有销售订单，所以这 5 笔订单添加到积压订单中。<p>
 * - 提交 2 笔销售订单，价格为 15 。没有采购订单的价格大于或等于 15 ，所以这 2 笔订单添加到积压订单中。<p>
 * - 提交 1 笔销售订单，价格为 25 。没有采购订单的价格大于或等于 25 ，所以这 1 笔订单添加到积压订单中。<p>
 * - 提交 4 笔采购订单，价格为 30 。前 2 笔采购订单与价格最低（价格为 15）的 2 笔销售订单匹配，从积压订单中删除这 2 笔销售订单。<p>
 * 第 3 笔采购订单与价格最低的 1 笔销售订单匹配，销售订单价格为 25 ，从积压订单中删除这 1 笔销售订单。<p>
 * 积压订单中不存在更多销售订单，所以第 4 笔采购订单需要添加到积压订单中。<p>
 * 最终，积压订单中有 5 笔价格为 10 的采购订单，和 1 笔价格为 30 的采购订单。所以积压订单中的订单总数为 6 。<p>
 * <p>
 * 示例 2：<p>
 * 输入：orders = [[7,1000000000,1],[15,3,0],[5,999999995,0],[5,1,1]]<p>
 * 输出：999999984<p>
 * 解释：输入订单后会发生下述情况：<p>
 * - 提交 10^9 笔销售订单，价格为 7 。没有采购订单，所以这 10^9 笔订单添加到积压订单中。<p>
 * - 提交 3 笔采购订单，价格为 15 。这些采购订单与价格最低（价格为 7 ）的 3 笔销售订单匹配，从积压订单中删除这 3 笔销售订单。<p>
 * - 提交 999999995 笔采购订单，价格为 5 。销售订单的最低价为 7 ，所以这 999999995 笔订单添加到积压订单中。<p>
 * - 提交 1 笔销售订单，价格为 5 。这笔销售订单与价格最高（价格为 5 ）的 1 笔采购订单匹配，从积压订单中删除这 1 笔采购订单。<p>
 * 最终，积压订单中有 (1000000000-3) 笔价格为 7 的销售订单，和 (999999995-1) 笔价格为 5 的采购订单。所以积压订单中的订单总数为 1999999991 ，等于 999999984 % (10^9 + 7) 。<p>
 * <p>
 * 提示：<p>
 * 1 <= orders.length <= 10^5<p>
 * orders[i].length == 3<p>
 * 1 <= pricei, amounti <= 10^9<p>
 * orderTypei 为 0 或 1<p>
 * <p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/number-of-orders-in-the-backlog">1801. 积压订单中的订单总数</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode1801 {

    @Test
    public void test01() {
        int[][] orders = {{10, 5, 0}, {15, 2, 1}, {25, 1, 1}, {30, 4, 0}};
//        int[][] orders = {{7, 1000000000, 1}, {15, 3, 0}, {5, 999999995, 0}, {5, 1, 1}};
        System.out.println(new Solution().getNumberOfBacklogOrders(orders));
    }
}

class Solution {
    public int getNumberOfBacklogOrders(int[][] orders) {
        /**
         * 每个 orders[i] = [pricei, amounti, orderTypei] 表示有 amounti 笔类型为 orderTypei 、价格为 pricei 的订单。
         * 订单类型 orderTypei 可以分为两种：
         * 0 表示这是一批采购订单 buy
         * 1 表示这是一批销售订单 sell
         *
         * 我们可以使用优先队列（大小根堆）维护当前的积压订单，
         * 其中大根堆 buy 维护积压的采购订单，
         * 小根堆 sell 维护积压的销售订单。
         * 堆中每个元素是一个二元组 (price, amount)，表示价格为 price 的订单数量为 amount。
         * 接下来，我们遍历订单数组 orders ，根据题意模拟即可。
         * 遍历结束后，我们将 buy 和 sell 中的订单数量相加，即为最终的积压订单数量。注意答案可能很大，需要对 10^9 + 7取模。
         *
         * 作者：lcbin
         * 链接：https://leetcode.cn/problems/number-of-orders-in-the-backlog/solution/by-lcbin-vf1s/
         * 来源：力扣（LeetCode）
         * 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
         */
        // 维护积压的采购订单
        PriorityQueue<int[]> buy = new PriorityQueue<>((a, b) -> b[0] - a[0]);
        // 维护积压的销售订单
        PriorityQueue<int[]> sell = new PriorityQueue<>((a, b) -> a[0] - b[0]);
        for (var e : orders) {
            int p = e[0], a = e[1], t = e[2];
            if (t == 0) {
                while (a > 0 && !sell.isEmpty() && sell.peek()[0] <= p) {
                    var q = sell.poll();
                    int x = q[0], y = q[1];
                    if (a >= y) {
                        a -= y;
                    } else {
                        sell.offer(new int[]{x, y - a});
                        a = 0;
                    }
                }
                if (a > 0) {
                    buy.offer(new int[]{p, a});
                }
            } else {
                while (a > 0 && !buy.isEmpty() && buy.peek()[0] >= p) {
                    var q = buy.poll();
                    int x = q[0], y = q[1];
                    if (a >= y) {
                        a -= y;
                    } else {
                        buy.offer(new int[]{x, y - a});
                        a = 0;
                    }
                }
                if (a > 0) {
                    sell.offer(new int[]{p, a});
                }
            }
        }
        long ans = 0;
        final int mod = (int) 1e9 + 7;
        while (!buy.isEmpty()) {
            ans += buy.poll()[1];
        }
        while (!sell.isEmpty()) {
            ans += sell.poll()[1];
        }
        return (int) (ans % mod);
    }
}
